from langchain.prompts import ChatPromptTemplate
from langchain.prompts.chat import SystemMessagePromptTemplate, HumanMessagePromptTemplate, AIMessagePromptTemplate

from langchain_ollama import ChatOllama

llm = ChatOllama(
    model="llama3.1",
    temperature=0.5
)

template = ChatPromptTemplate.from_messages([
    SystemMessagePromptTemplate.from_template("你是 {product} 的客服助手。你的名字叫{assistant_name}。请用中文回复以下问题："),
    HumanMessagePromptTemplate.from_template("你好，请问你是谁？"),
    AIMessagePromptTemplate.from_template("我很好，谢谢！"),
    HumanMessagePromptTemplate.from_template("{query}"),
])

prompt = template.format_messages(
    product="数易智能AI产品",
    assistant_name="Serena",
    query="请问你是谁？"
)

# 调用 LLM
result = llm.invoke(prompt)

print(result.content)